package leetcode.array;

/**
 * @author Cheng Jun
 * Description: 给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。
 * <p>
 * 不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
 * <p>
 * 元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/remove-element
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * @version 1.0
 * @date 2021/12/6 11:07
 * <p>
 * 输入：nums = [0,1,2,2,3,0,4,2], val = 2
 * 输出：5, nums = [0,1,4,0,3]
 */
public class removeElement {

    public static void main(String[] args) {
        System.out.println(removeElement(new int[]{1, 1, 3, 3}, 3));
        System.out.println(removeElement1(new int[]{1, 1, 3, 3}, 3));
    }

    // nums = [0,1,2,2,3,0,4,2], val = 2
    // 遍历数组，如果遇到 val，就移到数组最有一个非val的索引
    static int removeElement(int[] nums, int val) {
        outer:
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] == val) {
                // 找到数组最有一个非val的索引
                for (int j = nums.length - 1; j >= i; j--) {
                    // 找到
                    if (nums[j] != val) {
                        nums[i] = nums[j];
                        nums[j] = val;
                        break;
                    } else {
                        if (i == j) {
                            break outer;
                        }
                    }
                }
            }
        }
        int res = nums.length;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] == val) {
                res = i;
                break;
            }
        }
        return res;
    }

    // 官方解答
    // 可以使用双指针：右指针 right 指向当前将要处理的元素，左指针 left 指向下一个将要赋值的位置。
    // 如果右指针指向的元素不等于 val，它一定是输出数组的一个元素，我们就将右指针指向的元素复制到左指针位置，然后将左右指针同时右移；
    // 如果右指针指向的元素等于 val，它不能在输出数组里，此时左指针不动，右指针右移一位。
    static int removeElement1(int[] nums, int val) {
        int left = 0;
        for (int right = 0; right < nums.length; right++) {
            if (nums[right] != val) {
                nums[left++] = nums[right];
            }
        }
        return left;
    }


}
